from userManager import UserManager
from roleManager import RoleManager
from menuManager import MenuManager
from userRoleManager import UserRoleManager
from deleteRole import DeleteRole


def menu():
    """
    用户权限管理
    :return:
    """
    menu = """
    1.用户管理
    2.角色管理
    3.权限管理
    4.用户授权
    5.解除权限
    0.退出系统
    """
    print(menu)


if __name__ == '__main__':
    while True:
        menu()
        num = int(input("请输入你选择的数字："))
        if num == 1:
            print("欢迎进入用户管理")
            user_manger = UserManager()
            user_manger.operation()
        elif num == 2:
            print("欢迎进入角色管理")
            roleManager = RoleManager()
            roleManager.operation()
        elif num == 3:
            print("欢迎进入权限管理")
            menuManager = MenuManager()
            menuManager.operation()
        elif num == 4:
            print("欢迎进入用户授权")
            userRoleManager = UserRoleManager()
            userRoleManager.operator()
        elif num == 5:
            print("欢迎进入解除权限")
            delete_role = DeleteRole()
            delete_role.operation()
        elif num == 0:
            print("退出系统")
            break
        else:
            print("输入错误，请重新输入：")
            continue
